from django.shortcuts import render
from rest_framework import mixins
from rest_framework import status
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework_extensions.cache.mixins import CacheResponseMixin
# Create your views here.
from rest_framework.viewsets import ReadOnlyModelViewSet, GenericViewSet

from .models import Area

from .serializers import AreaSerializer,SubAreaSerializer
from rest_framework.decorators import action


class AreaViewSet(CacheResponseMixin,ReadOnlyModelViewSet):
    """
    行政区划信息
    """
    pagination_class=None # 区划信息不分页
    def get_queryset(self):

        """
        提供数据集
        """
        if self.action=='list':
            return Area.objects.filter(parent=None)
        else:
            return Area.objects.all()

    def get_serializer_class(self):

        """
        提供序列化器
        """
        if self.action=='list':
            return AreaSerializer
        else:
            return SubAreaSerializer

